package com.government.area.service.impl;

import java.util.HashMap;
import java.util.List;
import java.util.Objects;

import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.government.common.core.domain.BaseTreeVo;
import com.government.common.utils.DateUtils;
import com.government.common.utils.TreeUtils;
import com.government.tag.domain.TagType;
import org.apache.commons.compress.utils.Lists;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.government.area.mapper.AreaMapper;
import com.government.area.domain.Area;
import com.government.area.service.IAreaService;
import org.springframework.transaction.annotation.Transactional;

/**
 * 地区Service业务层处理
 * 
 * @author system
 * @date 2023-09-14
 */
@Service
@Transactional(rollbackFor = Exception.class)
public class AreaServiceImpl implements IAreaService 
{
    @Autowired
    private AreaMapper areaMapper;

    /**
     * 查询地区
     * 
     * @param id 地区主键
     * @return 地区
     */
    @Override
    public Area selectAreaById(String id)
    {
        return areaMapper.selectAreaById(id);
    }

    /**
     * 查询地区列表
     * 
     * @param area 地区
     * @return 地区
     */
    @Override
    public List<Area> selectAreaList(Area area)
    {
        return areaMapper.selectAreaList(area);
    }

    /**
     * 新增地区
     * 
     * @param area 地区
     * @return 结果
     */
    @Override
    public int insertArea(Area area)
    {
        area.setCreateTime(DateUtils.getNowDate());
        return areaMapper.insertArea(area);
    }

    /**
     * 修改地区
     * 
     * @param area 地区
     * @return 结果
     */
    @Override
    public int updateArea(Area area)
    {
        area.setUpdateTime(DateUtils.getNowDate());
        return areaMapper.updateArea(area);
    }

    /**
     * 批量删除地区
     * 
     * @param ids 需要删除的地区主键
     * @return 结果
     */
    @Override
    public int deleteAreaByIds(String[] ids)
    {
        return areaMapper.deleteAreaByIds(ids);
    }

    /**
     * 删除地区信息
     * 
     * @param id 地区主键
     * @return 结果
     */
    @Override
    public int deleteAreaById(String id)
    {
        return areaMapper.deleteAreaById(id);
    }

    /**
     * 查询地区树结构信息
     *
     * @return 部门树信息集合
     */
    @Override
    public List<BaseTreeVo> areaTree() {

        List<Area> areaList = areaMapper.selectList(Wrappers.<Area>lambdaQuery());
        if (CollectionUtils.isEmpty(areaList)) {

            return null;
        }
        HashMap<Long, List<BaseTreeVo>> allNodeMap = new HashMap<>();

        for (Area item : areaList) {
            Long parentId = Objects.isNull(item.getParentId()) ? 0 : item.getParentId();
            List<BaseTreeVo> nodeList = allNodeMap.get(parentId);
            if (CollectionUtils.isEmpty(nodeList)) {

                nodeList = Lists.newArrayList();
            }
            BaseTreeVo areaTreeVO = new BaseTreeVo();
            areaTreeVO.setId(String.valueOf(item.getId()));
            areaTreeVO.setValue(item.getId());
            areaTreeVO.setLabel(item.getName());
            nodeList.add(areaTreeVO);
            allNodeMap.put(parentId, nodeList);
        }
        List<BaseTreeVo> rootNode = allNodeMap.get(0L);

        return TreeUtils.buildTree(rootNode, allNodeMap);
    }
}
